package com.atguigu.springbootrabbitmq.config;

import lombok.extern.slf4j.Slf4j;
import org.springframework.amqp.core.Message;
import org.springframework.amqp.rabbit.connection.CorrelationData;
import org.springframework.amqp.rabbit.core.RabbitTemplate;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.stereotype.Component;

import javax.annotation.PostConstruct;

/**
 * 生产者连接交换机,成功或失败,都回调这个接口
 * 回调接口,消息接收成功或失败,都会调用这个接口
 * RabbitTemplate.ReturnCallback 回退接口
 *
 *
 *
 * 在生产环境中由于一些不明原因,导致RabbitMq重启,在RabbitMQ重启期间
 * 生产者投递失败,导致消息丢失,需要手动处理和回复。
 *
 * 下面的类RabbitTemplate.ConfirmCallback就是当消息投递到交换机,而回调的类。
 *  会通知生产者,消息到达的情况。
 *
 *  下面的类RabbitTemplate.ReturnCallback就是
 *      Mandatory参数
 *          在仅仅开启了生产者确认机制的情况下,交换机接受到消息后,
 *          会直接给消息生产者发送确认消息,如果发现该消息无法传递到队列，
 *          也就是无法路由,那么消息就会被直接丢弃，此时生产者是不知道消息被丢弃这个事件的。
 *
 *
 *
 *
 *
 *
 */
@Slf4j
@Component
public class MyCallBack implements RabbitTemplate.ConfirmCallback, RabbitTemplate.ReturnCallback {

    //注入
    @Autowired
    private RabbitTemplate rabbitTemplate;

    /**
     * 将当前类的对象注入到rabbitTemplate
     */
    @PostConstruct
    public void init() {
        rabbitTemplate.setConfirmCallback(this);
        rabbitTemplate.setReturnCallback(this);
    }


    /**
     * 交换机确认回调方法
     * 1.生产者发消息 交换机接收到了 回调
     * 参数1 correlationData 保存回调消息的ID及相关信息
     * 参数2 交换机收到消息 true
     * 参数3 cause 成功就是null
     *
     *
     * 2.发消息 交换机接收失败了 回调
     * 参数1 correlationData 保存回调消息的ID及相关信息
     * 参数2 交换机收到消息 ack = false
     * 参数3 cause 失败的原因
     *
     * @param correlationData
     * @param ack
     * @param cause
     */
    @Override
    public void confirm(CorrelationData correlationData, boolean ack, String cause) {
        String id = correlationData != null ? correlationData.getId() : "";

        if (ack) {
            //如果成功了ack为true
            log.info("交换机收到Id为: {}的消息", id);
        } else {
            //如果失败了
            log.info("交换机还未收到Id为:{}的消息,由于原因:{}", id, cause);
        }
    }


    /**
     * 在仅开启了生产者确认机制的情况下,交换机收到消息后,会直降给消息生产者发送确认消息,
     * 如果发现该消息不可路由,就是到达不了队列,那么消息会被直接丢弃,此时生产者是不知道消息被
     * 丢弃这个事件的。
     *
     *
     * 下面这个方法,可以在当消息传递过程中不可到达目的地时将消息返回给生产者
     * 只有不可达目的地的时候 才进行回退
     */
    @Override
    public void returnedMessage(Message message, int replyCode, String replyText, String exchange, String routingKey) {
        log.error("消息{},被交换机{}退回,退回原因:{},路由Key:{}", new String(message.getBody()), exchange, replyText, routingKey);
    }


}